from pydantic import BaseModel

# Define your data models here
class UserRegisterReq(BaseModel):
    username: str
    email: str
    password: str

class UserRegisterRsp(BaseModel):
    user_id: int
    username: str
    email: str
    message: str

class LoginReq(BaseModel):
    username: str
    password: str

class UserInfoRsp(BaseModel):
    user_id: int
    username: str
    email: str
    token: str